# 创建学生信息模型
class StudentInfo(object):
    """
    用于将StudentInfo.csv文件中的文本数据转换为对象
    """
    # 因为是基于csv文件存储数据,所以分隔符使用","
    def __init__(self, data: str, sep=","):
        data_list = data.split(sep)
        self.sid = data_list[0]
        self.name = data_list[1]
        self.gender = data_list[2]
        self.classid = data_list[3]

    def to_csv_str(self):
        """
        生成用于写入csv文件的文本数据
        :return: str
        """
        return ",".join((self.sid, self.name, self.gender, self.classid)) + "\n"

    # 重写str方法,用于输出学生信息
    def __str__(self):
        return f"学号:{self.sid}   姓名:{self.name}    性别:{self.gender}    班级号:{self.classid}"

    @staticmethod
    def get_csv_head():
        """
        csv文件标头
        :return: str
        """
        return "学号,姓名,性别,班级号\n"

